package Elementary_algorithm.Array;

/*
加一
给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外，这个整数不会以零开头。

示例1：
输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字 123。
示例2：
输入：digits = [4,3,2,1]
输出：[4,3,2,2]
解释：输入数组表示数字 4321。
示例 3：
输入：digits = [9]
输出：[1,0]
解释：输入数组表示数字 9。
加 1 得到了 9 + 1 = 10。
因此，结果应该是 [1,0]。

作者：LeetCode
链接：https://leetcode.cn/leetbook/read/top-interview-questions-easy/x2cv1c/
 */

import java.util.Arrays;

public class _07加一 {

    public static void main(String[] args) {
        int[] arr = {9,8,7,6,5,4,3,2,1};
        int[] ints = plusOne(arr);
        for (int i = 0; i < ints.length; i++) {
            System.out.println(ints[i]);
        }
    }

    //不知道为啥AC不了
    public static int[] plusOne(int[] digits) {
        int n = digits.length;
        int num = 0;
        for (int i = 0; i < n; i++) {
            num = num * 10 + digits[i];
        }
        num++;
        int[] res = new int[n + 1];
        int index = n;
        while (num > 0) {
            res[index--] = num % 10;
            num /= 10;
        }
        return res[0] != 1 ? Arrays.copyOfRange(res,1,n + 1) : res;
    }

    //官解：找出最长的后缀 9
    class Solution {
        public int[] plusOne(int[] digits) {
            int n = digits.length;
            for (int i = n - 1; i >= 0; --i) {
                if (digits[i] != 9) {
                    ++digits[i];
                    for (int j = i + 1; j < n; ++j) {
                        digits[j] = 0;
                    }
                    return digits;
                }
            }

            // digits 中所有的元素均为 9
            int[] ans = new int[n + 1];
            ans[0] = 1;
            return ans;
        }
    }
}
